#学生信息管理系统
import os
#定义一个主菜单
filename='student.txt'
def main():
    while True:
        menum()
        choice=int(input("请选择："))
        if choice in [0,1,2,3,4,5,6,7]:
            if choice==0:
                answer=input("你确定要退出系统吗？Y/n")
                if answer =='Y' or answer=='y':
                    print("谢谢使用！")
                    break  #退出系统
                else:
                    continue
            elif choice==1:
                insert()  #增加学生信息
            elif choice==2:
                search()#查找学生信息
            elif choice==3:
                delete()#删除学生信息
            elif choice==4:
                modify()#修改学生信息
            elif choice==5:
                sort()#排序
            elif choice==6:
                total()#统计学生信息总数
            elif choice==7:
                show()#显示学生信息
def menum():
    print("%%%%%%%%%%%%%%%学生信息管理系统%%%%%%%%%%%%%%")
    print("================功能菜单====================")
    print('\t\t\t\t\t\t\t\t1.录入学生信息')
    print('\t\t\t\t\t\t\t\t2.查找学生信息')
    print('\t\t\t\t\t\t\t\t3.删除学生信息')
    print('\t\t\t\t\t\t\t\t4.修改学生信息')
    print('\t\t\t\t\t\t\t\t5.排序')
    print('\t\t\t\t\t\t\t\t6.统计所有总人数')
    print('\t\t\t\t\t\t\t\t7.显示所有学生信息')
    print('\t\t\t\t\t\t\t\t0.退出系统')
#1插入学生信息
def insert():
    student_list=[]#存储学生信息
    while True:
        Id=input("请输入学生的ID（如1001）")
        if not Id:
            break
        Name=input("请输入学生姓名：")
        if not Name: #如果姓名为false则跳出循环
            break
        try:
            english=int(input("请输入英语成绩："))
            python=int(input("请输入python成绩："))
            java=int(input("请输入java成绩："))

        except:
            print("输入成绩无效，不是整数，请重新输入")
            continue
        #将录入的学生信息保存在字典中；
        student={'id':Id,"name":Name,'english':english,'python':python,'java':java}
        student_list.append(student)#将学生信息添加到列表中
        answer=input("是否继续录入学生信息？y/n\n")#/n表示换行
        if answer=='y' or answer=='Y':
            continue
        else:
            break
        #保存学生信息到文件中，
    # 调用save函数，保存学生信息
    save(student_list)
    print("学生信息保存完毕")
#保存学生信息
def save(lis):
    try:
        stu_txt=open(filename,'a',encoding="utf8")#以追加的形式打开
    except:
        stu_txt=open(filename,'w',encoding='utf8')#w以写入的形式打开
    for item in lis:
        stu_txt.write(str(item)+'\n')
    stu_txt.close()

#显示学生信息
def show_student(lst):
    if len(lst)==0:
        print('没有查询到学生信息。没有数据')
        return
    #定义标题显示格式
    format_title='{:^6}\t{:^12}\t{:^8}\t{:^10}\t{:^10}\t{:^8}'
    print(format_title.format('ID','姓名','英语成绩','python成绩','java成绩','总成绩'))
    #定义内容的显示格式
    format_data='{:^6}\t{:^12}\t{:^8}\t{:^10}\t{:^10}\t{:^8}'
    for item in lst:
        print(format_data.format(item.get('id'),
              item.get('name'),item.get('english'),item.get('python'),item.get('java'),
        int(item.get('english'))+int(item.get('python'))+int(item.get('java'))
              ))
#2查抄学生信息
def search():
    student_query=[]
    while True:
        id=''
        name=''
        if os.path.exists(filename):
            mode=input("按ID查找请输入1，按姓名查找请输入2：")
            if mode=='1':
                id=input('请输入学生的ID：')
            elif mode=='2':
                name=input('请输入学生的姓名：')
            else:
                print('您的输入有误，请重新输入')
                search()
            with open(filename,'r',encoding='utf-8') as rfile:
                student=rfile.readlines()
                for item in student:
                    d=dict(eval(item))
                    if id!='':
                        if d['id']==id:
                            student_query.append(d)
                    elif name!='':
                        if d['name']==name:
                            student_query.append(d)
            #显示查询结果
            show_student(student_query)
            student_query.clear()#清空列表
            answer=input('是否要继续进行查询操作？y/n')
            if answer=='y':
                continue
            else:
                break
        else:
            print('暂时未保存学生信息')
            return

#3从控制台录入学生id，到磁盘文件中，找到对应的学生信息 ，将其删除
def delete():
    while True:
        student_id=input("请输入要删除的学生ID：")
        if student_id!='':#判断如果输入的学生id不为空，就往下进行
            if os.path.exists(filename):#判断磁盘中当前目录下是否存在filename文件
                with open(filename,'r',encoding='utf-8') as file:
                    student_old=file.readlines()
            else:#如果不存在filename文件，就给student_old定义个空列表
                student_old=[]
            flag=False#标识符，标志是否删除的
            if student_old:
                #如果列表不为空，则读取文件
                with open(filename,'w',encoding="utf-8") as wfile:
                    d={}
                    for item in student_old:
                        d=dict(eval(item))#将字符串转换成字典
                        if d['id']!=student_id:
                            wfile.write(str(d)+'\n')
                        else:
                            flag=True
                    if flag:
                        print(f'id为{student_id}的学生的信息已删除！')
                    else:
                        print(f'没有找到ID为{student_id}的学生信息。')
            else:
                print('无此学生信息！')
                break
            show()#删除完后重新显示所有学生信息
            answer=input("是否继续删除学生信息？y/n")
            if answer=='y'or answer=='Y':
                continue
            else:
                break
#3修改学生信息
def modify():
    show()
    if os.path.exists(filename):
        with open(filename,'r',encoding='utf8') as rfile:
            student_old=rfile.readlines()
    else:
        return
    student_id=input("请输入要修改的学生的ID：")
    with open(filename,'w',encoding='utf8')as wfile:
        for item in student_old:
            d=dict(eval(item))
            if d['id']==student_id:
                print("找到了学生的信息，可以进行修改：")
                while True:
                    try:
                        d['name']=input("请输入姓名：")
                        d['english']=input("请输入英语成绩：")
                        d['python']=input("请输入python成绩：")
                        d['java']=input("请输入java成绩：")
                    except:
                        print("您的输入有误。请重新输入")
                    else:
                        break
                wfile.write(str(d)+'\n')
                print("修改成功！！！")
            else:
                wfile.write(str(d)+'\n')
        answer=input("是否继续修改其他学生信息？y/n\n")
        if answer=='y':
            modify()
#5排序
def sort():
    show()
    if os.path.exists(filename):
        with open(filename,'r',encoding='utf-8')as rfile:
            student_list=rfile.readlines()
        student_new=[]#将字符串转成字典类型存储到student_new中
        for item in student_list:
            d=dict(eval(item))
            student_new.append(d)
    else:
        return
    asc_or_desc=input("请悬着（0，升序；1，降序）")
    if asc_or_desc=='0':
        asc_or_desc=False
    elif asc_or_desc=='1':
        asc_or_desc=True
    else:
        print("你的输入有误")
        sort()
    mode=input("请选择排序方式：（1，按英语成绩；2，按python成绩；3，按java成绩;4,按总成绩排序）")
    if mode=='1':
        student_new.sort(key=lambda student_new:int(student_new['english']),reverse=asc_or_desc)
    elif mode=='2':
        student_new.sort(key=lambda student_new:int(student_new['python']),reverse=asc_or_desc)
    elif mode=='3':
        student_new.sort(key=lambda student_new:int(student_new['java']),reverse=asc_or_desc)
    elif mode=='4':
        student_new.sort(key=lambda student_new:int(student_new['english'])+int(student_new['python'])+int(student_new['java']),reverse=asc_or_desc)
    else:
        print("你的输入有误！请重新输入")
        sort()
    show_student(student_new)

#6学生信息总数
def total():
    if os.path.exists(filename):
        with open(filename,'r',encoding='utf-8')as rfile:
            students=rfile.readlines()
        if students:
            print(f'一共有{len(students)}名学生。')
        else:#有文件，但是里面没有数据
            print('还没有录入学生信息。')
    else:
        print("")

#7显示全部学生信息
def show():
    student_list=[]
    if os.path.exists(filename):
        with open(filename,'r',encoding='utf-8') as rfile:
            students=rfile.readlines()
            for item in students:
                student_list.append(eval(item))
            if student_list:
                show_student(student_list)
    else:
        print('暂时未保存过数据')

# if __name__=='__main__':
#      main()
if __name__ == '__main__':
    main()


